/* 异步日志缓冲区*/

#ifndef __M_BUFFER_H__
#define __M_BUFFER_H__

#include "util.hpp"
#include <vector>
#include <cassert>
using namespace std;

namespace log{

    #define DEFAULT_BUFFER_SIZE (1*1024*1024)
    #define THRESHOLD_BUFFER_SZIE (10*DEFAULT_BUFFER_SIZE)
    #define INCREAMENT_BUFFER_SIZE (5*DEFAULT_BUFFER_SIZE)

    class Buffer{
    public:
        Buffer():_buf(DEFAULT_BUFFER_SIZE), _ridx(0), _widx(0){

        }

        //向缓冲区写入数据
        void push(const char* data, size_t len){

            //1、检测缓冲区大小
            //缓冲区剩余空间不足：1.扩容 2.阻塞/返回false
            // if(len > writeableSize()){
            //     return;
            // }

            expansion(len);


            //2、将数据拷贝到缓冲区
            copy(data, data + len, &_buf[_widx]);

            //3、将当前写入位置向后移动
            moveWidx(len);
        }

        //获取可读数据的起始地址
        const char* getBegin(){
            //返回可读位置的地址
            return &_buf[_ridx];
        }

        //获取可读数据的长度
        size_t readableSize(){
            //可写位置-可读位置
            //由于使用的是双缓冲区，处理完就交换，和环形缓冲区的循环写入不一样，所以不存在空间的循环利用
            return _widx - _ridx;
        }

        //获取可写数据的长度
        size_t writeableSize(){
            //对于扩容方式来说，不存在可写数据长度，因为总是可写
            //这个接口是针对固定大小缓冲区的

            //末尾位置-可写位置
            return _buf.size() - _widx;
        }

        //向后移动ridx
        void moveRidx(size_t len){
            assert(len <= readableSize());
            _ridx += len;
        }

        //向后移动widx
        void moveWidx(size_t len){
            assert(len <= writeableSize());
            _widx += len;
        }

        //初始化/重置缓冲区
        void reset(){
            _ridx = _widx = 0;

        }

        //交换两个缓冲区
        void swapBuffer(Buffer &buf){
            _buf.swap(buf._buf);
            swap(_ridx, buf._ridx);
            swap(_widx, buf._widx);
        }

        //判断缓冲区是否为空
        bool empty(){
            return _ridx == _widx;
        }

    private:
        void expansion(size_t len){
            if(len <= writeableSize()){
                return;
            }
            size_t newsize = 0;
            //小于阈值指数增长，大于阈值线性增长
            if(_buf.size() < THRESHOLD_BUFFER_SZIE){
                //每次都+len，防止扩容后还是不够
                newsize = _buf.size() * 2 + len;
            }else{
                newsize = _buf.size() + INCREAMENT_BUFFER_SIZE + len;
            }
            _buf.resize(newsize);
        }

    private:
        vector<char> _buf;
        size_t _ridx;       //当前可读数据的下标
        size_t _widx;       //当前可写数据的下标
    };
}

#endif